package dfs;

import tree.treenode.TreeNode;

/**
 * @BelongsProject: LeetCode
 * @BelongsPackage: com.elvis.leetcode.dfs
 * @Author: Elvis
 * @CreateTime: 2019-05-27 12:19
 * Description:
 * 给定一个二叉树，它的每个结点都存放一个 0-9 的数字，每条从根到叶子节点的路径都代表一个数字。
 *
 * 例如，从根到叶子节点路径 1->2->3 代表数字 123。
 *
 * 计算从根到叶子节点生成的所有数字之和。
 *
 * 说明: 叶子节点是指没有子节点的节点。
 *
 * 示例 1:
 *
 * 输入: [1,2,3]
 *     1
 *    / \
 *   2   3
 * 输出: 25
 * 解释:
 * 从根到叶子节点路径 1->2 代表数字 12.
 * 从根到叶子节点路径 1->3 代表数字 13.
 * 因此，数字总和 = 12 + 13 = 25.
 * 示例 2:
 *
 * 输入: [4,9,0,5,1]
 *     4
 *    / \
 *   9   0
 *  / \
 * 5   1
 * 输出: 1026
 * 解释:
 * 从根到叶子节点路径 4->9->5 代表数字 495.
 * 从根到叶子节点路径 4->9->1 代表数字 491.
 * 从根到叶子节点路径 4->0 代表数字 40.
 * 因此，数字总和 = 495 + 491 + 40 = 1026.
 */
public class Leetcode129 {

    /**
     * self_done
     * @param root
     * @return
     */
    public int sumNumbers(TreeNode root) {
        if (root == null) return 0;
        return dfs(root, 0);
    }
    private int dfs(TreeNode root, int sum){
        if (root.left == null && root.right == null){
            sum += root.val;
            return sum;
        }
        sum += root.val;
        sum *= 10;
        if (root.left != null && root.right != null){
            return dfs(root.left, sum) + dfs(root.right, sum);
        }
        if (root.left != null){
            return dfs(root.left, sum);
        }
        if (root.right != null){
            return dfs(root.right, sum);
        }
        return -1;
    }

    public static void main(String[] args) {
        /*TreeNode node = new TreeNode(4);
        node.right = new TreeNode(0);
        node.left = new TreeNode(9);
        node.left.left = new TreeNode(5);
        node.left.right = new TreeNode(1);*/
        TreeNode node = new TreeNode(0);
        node.left = new TreeNode(1);
        new Leetcode129().sumNumbers(node);
    }
}
